package com.jhc.yolo.pojo;

import java.util.List;

/**
 * 识别配置
 */
public class ModelConfig {
    private String modelName;
    private String modelPath;
    private List<String> labelNames;
    // 识别阈值
    private float confThreshold;
    private float nmsThreshold;
    private int gpuDeviceId;


    public String getModelName() {
        return modelName;
    }

    public void setModelName(String modelName) {
        this.modelName = modelName;
    }

    public String getModelPath() {
        return modelPath;
    }

    public void setModelPath(String modelPath) {
        this.modelPath = modelPath;
    }

    public List<String> getLabelNames() {
        return labelNames;
    }

    public void setLabelNames(List<String> labelNames) {
        this.labelNames = labelNames;
    }

    public float getConfThreshold() {
        return confThreshold;
    }

    public void setConfThreshold(float confThreshold) {
        this.confThreshold = confThreshold;
    }

    public float getNmsThreshold() {
        return nmsThreshold;
    }

    public void setNmsThreshold(float nmsThreshold) {
        this.nmsThreshold = nmsThreshold;
    }

    public int getGpuDeviceId() {
        return gpuDeviceId;
    }

    public void setGpuDeviceId(int gpuDeviceId) {
        this.gpuDeviceId = gpuDeviceId;
    }
}
